import cv2


class Reward:
    def __init__(self):
        pass

    def get_reward(self, state):
        return 0

    def get_loc_reward(self, state):
        _, v, _, l = cv2.minMaxLoc(cv2.matchTemplate(state, cv2.imread(r""), method=cv2.TM_CCOEFF_NORMED))
        if v > 0.95: return l[0]
        return 0
